package com.lx.servlet;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.UUID;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.lx.bean.FileTest;
import com.lx.dao.FileTestImpl;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

/*
* 上传文件接口：
响应标准 http 协议的 post 请求发送来的文件，接收到文件后将文件重命名
（使用 UUID）并按照日期保存至服务器文件系统的不同的目录中（目录格
式 yyyyMMdd） ，同时将文件大小、文件类型，原始文件名、创建时间、文
件保存目录地址等元数据记录至数据库中，同时将 UUID 返回给客户端。
* */
@WebServlet("/UploadHandleServlet")
public class UploadHandleServlet extends HttpServlet {

    public void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        //创建时间转换器SimpleDateFormat
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
        String times = sdf.format(new Date());
        //得到上传文件的保存目录，将上传的文件存放于WEB-INF目录下，不允许外界直接访问，保证上传文件的安全
        String savePath = this.getServletContext().getRealPath("/WEB-INF/upload/"+times);
        String savePath2 = this.getServletContext().getRealPath("/WEB-INF/upload");
        File file1 = new File(savePath2);
        //判断上传文件的保存目录是否存在
        if (!file1.exists() && !file1.isDirectory()) {
            System.out.println(savePath2+"目录不存在，需要创建");

            //创建目录
            file1.mkdir();
            File file= new File(savePath);
            file.mkdir();

        }
        //消息提示
        String message = "";
        try{
            //使用Apache文件上传组件处理文件上传步骤：
            //1、创建一个DiskFileItemFactory工厂
            DiskFileItemFactory factory = new DiskFileItemFactory();
            //2、创建一个文件上传解析器
            ServletFileUpload upload = new ServletFileUpload(factory);
            //解决上传文件名的中文乱码
            upload.setHeaderEncoding("UTF-8");
            //3、判断提交上来的数据是否是上传表单的数据
            if(!ServletFileUpload.isMultipartContent(request)){
                //按照传统方式获取数据
                return;
            }
            //4、使用ServletFileUpload解析器解析上传数据，解析结果返回的是一个List<FileItem>集合，每一个FileItem对应一个Form表单的输入项
            List<FileItem> list = upload.parseRequest(request);
            for(FileItem item : list){
                //如果fileitem中封装的是普通输入项的数据
                if(item.isFormField()){
                    String name = item.getFieldName();
                    //解决普通输入项的数据的中文乱码问题
                    String value = item.getString("UTF-8");
                    //value = new String(value.getBytes("iso8859-1"),"UTF-8");
                    System.out.println(name + "=" + value);
                }else{//如果fileitem中封装的是上传文件

                    String uuid = UUID.randomUUID().toString().replaceAll("-", "");
                    //将 UUID 返回给客户端

                    request.setAttribute("uuid",uuid);
                    // 获得文件原始名称
                    String fileOldName = item.getName();
                    // 获得文件后缀名称
                    String suffixName = fileOldName.substring(fileOldName.lastIndexOf(".") + 1).toLowerCase();
                    // 生成最新的uuid文件名称
                    String filename = uuid + "."+ suffixName;

                    ////元数据记录至数据库中
                    FileTest fileTest = new FileTest();
                    //设置uuid
                    fileTest.setUuid(uuid);
                    //大小转换成kb，设置文件大小
                    long size = item.getSize();
                    fileTest.setFileSize(size/1024+"kb");
                    //设置文件类型
                    fileTest.setFileType(suffixName);
                    //设置园诗名称
                    fileTest.setFileName(fileOldName);
                    //设置创建时间
                    fileTest.setCreateTime(times);
                    //设置保存路径
                    fileTest.setFileDirectory(savePath);
                    //创建FileTestImpl对象，把元数据添加到mysql数据库
                    FileTestImpl fileTest1 = new FileTestImpl();
                    fileTest1.addFile(fileTest);
                    //String filename = item.getName();
                    System.out.println(filename);
                    if(filename==null || filename.trim().equals("")){
                        continue;
                    }

                    //处理获取到的上传文件的文件名的路径部分，只保留文件名部分
                    filename = filename.substring(filename.lastIndexOf("\\")+1);
                    //获取item中的上传文件的输入流
                    InputStream in = item.getInputStream();
                    //创建一个文件输出流
                    FileOutputStream out = new FileOutputStream(savePath + "\\" + filename);
                    //创建一个缓冲区
                    byte buffer[] = new byte[1024];
                    //判断输入流中的数据是否已经读完的标识
                    int len = 0;
                    //循环将输入流读入到缓冲区当中，(len=in.read(buffer))>0就表示in里面还有数据


                    while((len=in.read(buffer))>0){
                        //使用FileOutputStream输出流将缓冲区的数据写入到指定的目录(savePath + "\\" + filename)当中
                        out.write(buffer, 0, len);
                    }
                    //关闭输入流
                    in.close();
                    //关闭输出流
                    out.close();
                    //删除处理文件上传时生成的临时文件
                    item.delete();
                    message = "文件上传成功！";
                }


            }
        }catch (Exception e) {
            message= "文件上传失败！";
            e.printStackTrace();

        }
        request.setAttribute("message",message);
        request.getRequestDispatcher("/message.jsp").forward(request, response);
    }

    public void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {

        doGet(request, response);
    }
}